absl-py>=0.12.0
flax>=0.3.3
jax>=0.2.12
jaxlib>=0.1.65
numpy>=1.19.5
tensorboard>=2.4.1
tensorflow>=2.5.0rc0
typing-extensions>=3.7.4.3
